n_gpu = 4 command = f'dlojz.py --image-size {image_size} -p 2 --test --deepspeed --deepspeed_config ds_config.json' jobid = gpu_jobs_submitter(command, n_gpu, MODULE, name=name, account=account, time_max='00:20:00', constraint='v100-32g') print(f'jobid = {jobid}')